package com.cloudkinto.common.common;

import java.io.*;
import java.nio.file.Files;
import java.util.Base64;

/**
 * @author long
 */
public class Base64AndFileStreamUtil {

    /**
     * 文件流转换成Base4
     */
    public static String fileToBase64(String path) {
        String base64 = null;
        InputStream in = null;
        try {
            File file = new File(path);
            in = Files.newInputStream(file.toPath());
            byte[] bytes = new byte[(int) file.length()];
            in.read(bytes);
            base64 = Base64.getEncoder().encodeToString(bytes);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (in != null) {
                try {
                    in.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return base64;
    }

    /**
     * Base64解码成File文件
     */
    public static InputStream base64ToFile(String base64, String filePath, String fileName) throws IOException {
        File file;
        //创建文件目录
        File dir = new File(filePath);
        if (!dir.exists() && !dir.isDirectory()) {
            dir.mkdirs();
        }
        BufferedOutputStream bos = null;
        FileOutputStream fos = null;
        byte[] bytes = Base64.getDecoder().decode(base64);
        file = new File(filePath + File.separator + fileName);
        try {
            fos = new FileOutputStream(file);
            bos = new BufferedOutputStream(fos);
            bos.write(bytes);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (bos != null) {
                try {
                    bos.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (fos != null) {
                try {
                    fos.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return Files.newInputStream(file.toPath());
    }


    public static InputStream base64ToFile(String base64) throws IOException {
        byte[] bytes = Base64.getDecoder().decode(base64);
        InputStream stream = new ByteArrayInputStream(bytes);

        return stream;
    }
}
